package com.hghivln.erpwarehouse.system.common.lang;

import com.hghivln.erpwarehouse.system.common.lang.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * 把没有层级关系的集合变成有层级关系的集合
 */
public class TreeNodeBuilder {

    public static List<TreeNode> build(List<TreeNode> treeNodes, Integer topPid) {
        ArrayList<TreeNode> nodes = new ArrayList<>();
        for (TreeNode node1 : treeNodes) {
            if (node1.getPid().equals(topPid)) {
                nodes.add(node1);
            }
            for (TreeNode node2 : treeNodes) {
                if (node1.getId().equals(node2.getPid())) {
                    node1.getChild().add(node2);
                }
            }
        }
        return nodes;
    }
}
